#include "gtest/gtest.h"

#include "DataStructure/Input/Element/ElementActiveDOFs.h"

TEST (GetEleActiveDofs, BeamElementTest)
{
    auto dofsTest = getElementActiveDofs (SG::DataStructure::FEM::ElementTypes::ELETYPE_Beam);

    auto dofsTarget =
        SG::DataStructure::FEM::DofSet{ SG::DataStructure::FEM::DOFDirections::DOF_Ux,   SG::DataStructure::FEM::DOFDirections::DOF_Uy,
                                        SG::DataStructure::FEM::DOFDirections::DOF_Uz,   SG::DataStructure::FEM::DOFDirections::DOF_Rotx,
                                        SG::DataStructure::FEM::DOFDirections::DOF_Roty, SG::DataStructure::FEM::DOFDirections::DOF_Rotz };

    ASSERT_TRUE (dofsTest == dofsTarget);
}